from OCC.Core.BRepPrimAPI import BRepPrimAPI_MakeWedge
from OCC.Core.gp import gp_Pnt, gp_Ax2, gp_Dir

my_wedge = BRepPrimAPI_MakeWedge(gp_Ax2(gp_Pnt(4, 0, 4), gp_Dir(1, 1, 0)), 4, 4, 4, 2, 2, 2, 2).Shape()
my_wedge = BRepPrimAPI_MakeWedge(gp_Ax2(gp_Pnt(-4, -8, -4), gp_Dir(0, 0, 1)), 4, 4, 8, 1, 1, 3, 7).Shape()
my_wedge = BRepPrimAPI_MakeWedge(gp_Ax2(gp_Pnt(2, -6, -4), gp_Dir(0, 0, 1)), 2, 3, 4, 1).Shape()

if __name__ == '__main__':
    from OCC.Display.SimpleGui import init_display

    display, start_display, add_menu, add_function_to_menu = init_display()
    display.DisplayShape(my_wedge, update=True)
    start_display()
